package com.atguigu.tingshu.user.service.impl;


import cn.hutool.core.date.DateUtil;
import cn.hutool.core.util.IdUtil;
import com.atguigu.tingshu.common.constant.RedisConstant;
import com.atguigu.tingshu.common.constant.SystemConstant;
import com.atguigu.tingshu.common.rabbit.constant.MqConst;
import com.atguigu.tingshu.common.rabbit.service.RabbitService;
import com.atguigu.tingshu.common.util.MongoUtil;
import com.atguigu.tingshu.model.user.UserListenProcess;
import com.atguigu.tingshu.user.service.UserListenProcessService;
import com.atguigu.tingshu.vo.album.TrackStatMqVo;
import com.atguigu.tingshu.vo.user.UserListenProcessVo;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Sort;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Service;

import java.math.BigDecimal;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.TimeUnit;

@Service
@SuppressWarnings({"all"})
public class UserListenProcessServiceImpl implements UserListenProcessService {

    @Autowired
    private MongoTemplate mongoTemplate;

    @Autowired
    private RedisTemplate redisTemplate;

    @Autowired
    private RabbitService rabbitService;

    /**
     * 需求一：添加/更新用户播放记录至 mongodb
     * 需求二：统计用户的播放记录
     *
     * @param trackStatMqVo
     * @param userId
     */
    @Override
    public void updateListenProcess(UserListenProcessVo userListenProcessVo, Long userId) {
        // 需求一：添加/更新用户播放记录至 mongodb
        // 从mongodb中获取用户播放记录，根据userId、trackId
        Query query = new Query();
        query.addCriteria(Criteria.where("userId").is(userId)
                .and("trackId").is(userListenProcessVo.getTrackId())
                .and("isShow").is(1));
        // 限制查询数量 避免前端请求混乱
        query.limit(1);
        // 以 userId 作为分区条件，创建 mongodbs
        UserListenProcess userListenProcessResult = mongoTemplate.findOne(query, UserListenProcess.class, MongoUtil.getCollectionName(MongoUtil.MongoCollectionEnum.USER_LISTEN_PROCESS, userId));
        // 判断用户播放记录是否存在
        if (userListenProcessResult == null) {
            // 不存在即为新增，即没有_id
            UserListenProcess userListenProcess = new UserListenProcess();
            userListenProcess.setUserId(userId);
            userListenProcess.setAlbumId(userListenProcessVo.getAlbumId());
            userListenProcess.setTrackId(userListenProcessVo.getTrackId());
            userListenProcess.setBreakSecond(userListenProcessVo.getBreakSecond());
            userListenProcess.setIsShow(1);
            userListenProcess.setCreateTime(new Date());
            userListenProcess.setUpdateTime(new Date());
            mongoTemplate.save(userListenProcess, MongoUtil.getCollectionName(MongoUtil.MongoCollectionEnum.USER_LISTEN_PROCESS, userId));
        } else {
            // 存在为更新，即有_id 直接刷新进度时间(但是此时需要获取进度时间)
            userListenProcessResult.setBreakSecond(userListenProcessVo.getBreakSecond());
            userListenProcessResult.setUpdateTime(new Date());
            mongoTemplate.save(userListenProcessResult, MongoUtil.getCollectionName(MongoUtil.MongoCollectionEnum.USER_LISTEN_PROCESS, userId));
        }

        // 需求二：统计用户的播放记录  albumStat trackStat
        //       统计专辑次数为一张专辑播放一次当天只统计该专辑一次(采用分布式锁作为标识，并设置过期时间为今天)
        // 设置业务标识，必须为唯一标识不可重复

        String flagKey = RedisConstant.USER_TRACK_REPEAT_STAT_PREFIX + userId + ":" + userListenProcessVo.getTrackId();
        Long ttl = DateUtil.endOfDay(new Date()).getTime() - System.currentTimeMillis();
        // 防止重复记录专辑音频播放次数
        Boolean flag = redisTemplate.opsForValue().setIfAbsent(flagKey, userId, ttl, TimeUnit.MILLISECONDS);
        if(flag){
            TrackStatMqVo  trackStatMqVo = new TrackStatMqVo();
            trackStatMqVo.setBusinessNo(IdUtil.getSnowflake().nextIdStr());
            trackStatMqVo.setAlbumId(userListenProcessVo.getAlbumId());
            trackStatMqVo.setTrackId(userListenProcessVo.getTrackId());
            trackStatMqVo.setStatType(SystemConstant.TRACK_STAT_PLAY);
            trackStatMqVo.setCount(1);
            // 发送更新播放记录消息
            rabbitService.sendMessage(MqConst.EXCHANGE_TRACK,MqConst.ROUTING_TRACK_STAT_UPDATE,trackStatMqVo);
        }

    }

    @Override
    public BigDecimal getTrackBreakSecond(Long trackId, Long userId) {
        // 从mongodb中获取用户播放记录，根据userId、trackId(与update联合)
        Query query = new Query();
        query.addCriteria(Criteria.where("userId").is(userId)
                .and("trackId").is(trackId)
                .and("isShow").is(1));
        UserListenProcess userListenProcess = mongoTemplate.findOne(query, UserListenProcess.class, MongoUtil.getCollectionName(MongoUtil.MongoCollectionEnum.USER_LISTEN_PROCESS, userId));
        if (userListenProcess != null) {
            return userListenProcess.getBreakSecond();
        }
        return new BigDecimal(0);
    }

    /**
     * 获取用户最近播放的音频
     * @param userId
     * @return
     */
    @Override
    public Map<String, Long> getLatelyTrack(Long userId) {
        HashMap<String, Long> mapResult = new HashMap<>();
        // 查询条件
        Query query = new Query();
        query.addCriteria(Criteria.where("userId").is(userId));
        query.with(Sort.by(Sort.Direction.DESC, "updateTime"));
        query.limit(1);
        // 查询结果
        UserListenProcess userListenProcess = mongoTemplate.findOne(query, UserListenProcess.class, MongoUtil.getCollectionName(MongoUtil.MongoCollectionEnum.USER_LISTEN_PROCESS, userId));
        if(userListenProcess != null){
            mapResult.put("albumId", userListenProcess.getAlbumId());
            mapResult.put("trackId", userListenProcess.getTrackId());
            return mapResult;
        }

        return null;

    }
}
